#!/bin/bash  
  
# 激活虚拟环境  
# source myenv/bin/activate  
#conda activate xty
  
# 设置数据目录路径  
DATA_PATH="/home/u202332803005/data"  
  
# 训练基线模型  
# echo "Training Baseline Model..."  
# CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py \  
#     -a resnet50 -b 64 --test-batch-size 256 --iters 200 --lr 3.5e-4 --epoch 70 \  
#     --dataset_src1 msmt17v1 --dataset_src2 cuhk03 --dataset_src3 market1501 -d dukemtmc \  
#     --logs-dir logs/Baseline \  
#     --data-dir $DATA_PATH  
  
# 训练IL模型  
echo "Training IL Model..."  
CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py \  
    -a resnet50 -b 64 --test-batch-size 256 --iters 200 --lr 3.5e-4 --epoch 70 \  
    --dataset_src1 msmt17v1 --dataset_src2 cuhk03 --dataset_src3 market1501 -d dukemtmc \  
    --logs-dir logs/IL \  
    --updateStyle \  
    --data-dir $DATA_PATH  
  
# # 退出虚拟环境（可选）  
# deactivate
